from openai import OpenAI
from dotenv import load_dotenv
import os

def qianwen_embedding(text, dim=512):
    # 获取.env文件的配置
    load_dotenv('./.env')
    QIANWEN_API_KEY = os.getenv('QIANWEN_API_KEY')
    # print(QIANWEN_API_KEY)
    client = OpenAI(
        api_key=QIANWEN_API_KEY,  # 如果您没有配置环境变量，请在此处用您的API Key进行替换
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"  # 百炼服务的base_url
    )

    completion = client.embeddings.create(
        model="text-embedding-v4",
        input=text,
        dimensions=dim,  # 指定向量维度（仅 text-embedding-v3及 text-embedding-v4支持该参数）
        encoding_format="float"
    )
    return completion.data[0].embedding

if __name__ == '__main__':
    # print(qianwen_embedding('衣服的质量杠杠的，很漂亮，不枉我等了这么久啊，喜欢，以后还来这里买'))
    print(qianwen_embedding('阿福招牌'))
